package ai.example.langchain4j.service;

import dev.langchain4j.store.embedding.EmbeddingMatch;
import dev.langchain4j.data.segment.TextSegment;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.List;

public interface LangChain4jService {

    /**
     * 上传单个或多个文件并生成向量
     * @param file 上传文件
     * @return 上传结果描述
     */
    String upload(MultipartFile file) throws IOException;

    /**
     * 搜索向量库
     * @param message 查询内容
     * @return 匹配的向量结果
     */
    List<EmbeddingMatch<TextSegment>> search(String message);
}
